Loading and plotting data

Loading and plotting data#

This notebook showcases how to load and plot data that was previously generated using the creator.py script.

%load_ext autoreload
%autoreload 2

%matplotlib inline

The autoreload extension is already loaded. To reload it, use:
  %reload_ext autoreload

import xarray as xr
import matplotlib as mpl

from causaldynamics.scm import create_scm_graph
from causaldynamics.plot import animate_3d_trajectories, plot_trajectories, plot_scm

All relevant data is stored in a single dataset for easy access. Let’s load it and look at the content:

The data is stored in the Data variables. To get the data, e.g. the time_series, you can index it like a dictionary.

Let’s plot the time series.

root_nodes = ds["root_nodes"]
plot_trajectories(time_series, root_nodes=root_nodes, sharey=False, node_color="orange", root_node_color="dimgrey")

<xarray.plot.facetgrid.FacetGrid at 0x1083349d0>

../_images/93ed296f4a7ca3ea22859e23d5cd4e618ed896e57c41e8f91a0aec01dd7718af.png

The adjacency matrix encodes the SCM graph structure. The first dimension are the incoming nodes and the second dimension are the outgoing nodes.

# You can access the underlying numpy array directly by calling .data
print(A.data)
# You can also convert it to a pandas DataFrame
print(A.to_dataframe())

[[0. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0.]
 [1. 0. 0. 0. 0.]
 [0. 0. 0. 1. 0.]]
                  adjacency_matrix
node_in node_out                  
0       0                      0.0
        1                      0.0
        2                      0.0
        3                      0.0
        4                      0.0
1       0                      1.0
        1                      0.0
        2                      0.0
        3                      0.0
        4                      0.0
2       0                      1.0
        1                      0.0
        2                      0.0
        3                      0.0
        4                      0.0
3       0                      1.0
        1                      0.0
        2                      0.0
        3                      0.0
        4                      0.0
4       0                      0.0
        1                      0.0
        2                      0.0
        3                      1.0
        4                      0.0

Let’s visualize the graph

G = create_scm_graph(A.data)
plot_scm(G, root_nodes=root_nodes)

(<Axes: title={'center': 'Structural Causal Model'}>,
 {0: array([ 0.19040298, -0.4336854 ]),
  1: array([0.74426205, 0.41151592]),
  2: array([0.6018516 , 0.03802172]),
  3: array([-0.53651663, -0.54202374]),
  4: array([-1.        ,  0.52617151])})

../_images/c811f66319c16a206bb847104d610c942db39ceefe163d0c5307562e7fcb8bd5.png

# Creating an animation of the trajectories. 
# This may take a while to run...
mpl.rcParams["animation.embed_limit"] = 50 * 1024**2  # Increase the limit to 50MB

animate_3d_trajectories(time_series, 
                        root_nodes=root_nodes, 
                        plot_type="subplots",
                        frame_skip=5, 
                        rotation_speed=0.2, 
                        rotate=True, 
                        show_history=True, 
                        save_path=None, 
                        return_html_anim=True, # perfect for use in notebooks :)
                        show_plot=False,
                        root_node_alpha=0.5,
                        node_alpha=0.5,
                        linewidth=1.5)